import redis
import Customer
import pickle

pool = redis.ConnectionPool(host='127.0.0.1', port=6379, db=1)
r = redis.Redis(connection_pool=pool)
# r.set('name', 'zhangsan')   # 添加
# print(r.get('name'))   # 获取


# 这个用做redis数据库初始化
def init_day_account():
    r.delete('day1_coin_account')
    r.delete('day2_coin_account')
    r.delete('day3_coin_account')
    r.delete('day1_coin')
    r.delete('day2_coin')
    r.delete('day3_coin')

    r.incrby("day1_coin_account", 50000)
    r.incrby("day2_coin_account", 100000)
    r.incrby("day3_coin_account", 350000)
    r.incrby("day1_coin", 10)
    r.incrby("day2_coin", 20)
    r.incrby("day3_coin", 50)

    r.incrby('hour8_coin_out1', 0)
    r.incrby('hour9_coin_out1', 0)
    r.incrby('hour10_coin_out1', 0)
    r.incrby('hour11_coin_out1', 0)
    r.incrby('hour12_coin_out1', 0)
    r.incrby('hour13_coin_out1', 0)
    r.incrby('hour14_coin_out1', 0)
    r.incrby('hour15_coin_out1', 0)
    r.incrby('hour16_coin_out1', 0)
    r.incrby('hour17_coin_out1', 0)
    r.incrby('hour18_coin_out1', 0)
    r.incrby('hour19_coin_out1', 0)
    r.incrby('hour20_coin_out1', 0)
    r.incrby('hour21_coin_out1', 0)
    r.incrby('hour22_coin_out1', 0)
    r.incrby('hour23_coin_out1', 0)

    r.incrby('hour8_coin_out2', 0)
    r.incrby('hour9_coin_out2', 0)
    r.incrby('hour10_coin_out2', 0)
    r.incrby('hour11_coin_out2', 0)
    r.incrby('hour12_coin_out2', 0)
    r.incrby('hour13_coin_out2', 0)
    r.incrby('hour14_coin_out2', 0)
    r.incrby('hour15_coin_out2', 0)
    r.incrby('hour16_coin_out2', 0)
    r.incrby('hour17_coin_out2', 0)
    r.incrby('hour18_coin_out2', 0)
    r.incrby('hour19_coin_out2', 0)
    r.incrby('hour20_coin_out2', 0)
    r.incrby('hour21_coin_out2', 0)
    r.incrby('hour22_coin_out2', 0)
    r.incrby('hour23_coin_out2', 0)

    r.incrby('hour8_coin_out3', 0)
    r.incrby('hour9_coin_out3', 0)
    r.incrby('hour10_coin_out3', 0)
    r.incrby('hour11_coin_out3', 0)
    r.incrby('hour12_coin_out3', 0)
    r.incrby('hour13_coin_out3', 0)
    r.incrby('hour14_coin_out3', 0)
    r.incrby('hour15_coin_out3', 0)
    r.incrby('hour16_coin_out3', 0)
    r.incrby('hour17_coin_out3', 0)
    r.incrby('hour18_coin_out3', 0)
    r.incrby('hour19_coin_out3', 0)
    r.incrby('hour20_coin_out3', 0)
    r.incrby('hour21_coin_out3', 0)
    r.incrby('hour22_coin_out3', 0)
    r.incrby('hour23_coin_out3', 0)


# 删除发币记录, 测试完使用的方法
def delete_data():
    r.delete('hour8_coin_out1')
    r.delete('hour9_coin_out1')
    r.delete('hour10_coin_out1')
    r.delete('hour11_coin_out1')
    r.delete('hour12_coin_out1')
    r.delete('hour13_coin_out1')
    r.delete('hour14_coin_out1')
    r.delete('hour15_coin_out1')
    r.delete('hour16_coin_out1')
    r.delete('hour17_coin_out1')
    r.delete('hour18_coin_out1')
    r.delete('hour19_coin_out1')
    r.delete('hour20_coin_out1')
    r.delete('hour21_coin_out1')
    r.delete('hour22_coin_out1')
    r.delete('hour23_coin_out1')

    r.delete('hour8_coin_out2')
    r.delete('hour9_coin_out2')
    r.delete('hour10_coin_out2')
    r.delete('hour11_coin_out2')
    r.delete('hour12_coin_out2')
    r.delete('hour13_coin_out2')
    r.delete('hour14_coin_out2')
    r.delete('hour15_coin_out2')
    r.delete('hour16_coin_out2')
    r.delete('hour17_coin_out2')
    r.delete('hour18_coin_out2')
    r.delete('hour19_coin_out2')
    r.delete('hour20_coin_out2')
    r.delete('hour21_coin_out2')
    r.delete('hour22_coin_out2')
    r.delete('hour23_coin_out2')

    r.delete('hour8_coin_out3')
    r.delete('hour9_coin_out3')
    r.delete('hour10_coin_out3')
    r.delete('hour11_coin_out3')
    r.delete('hour12_coin_out3')
    r.delete('hour13_coin_out3')
    r.delete('hour14_coin_out3')
    r.delete('hour15_coin_out3')
    r.delete('hour16_coin_out3')
    r.delete('hour17_coin_out3')
    r.delete('hour18_coin_out3')
    r.delete('hour19_coin_out3')
    r.delete('hour20_coin_out3')
    r.delete('hour21_coin_out3')
    r.delete('hour22_coin_out3')
    r.delete('hour23_coin_out3')
    r.flushdb()


if __name__ == '__main__':
    delete_data()
    init_day_account()
    # test_cus()

